import sys,os
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__),'..'))
sys.path.append(project_root)
import witin_nn
print('Env test:\nPytorch:')
import torch
print(torch.cuda.is_available())
print('Witin:')

from witin_nn.interface import ConfigFactory

a = ConfigFactory.GlobalConfigFactory()
print(a.get_dafault_config())
b=witin_nn.nn.conv2d.Conv2d(3, 64, kernel_size=7, stride=2, padding=3, bias=False)
print(b)